# -*- coding: utf-8 -*-
from scpy.logger import get_logger
import os
import sys
from pgutil import PgUtil
from recruitment_ana_util import search_category_type

reload(sys)
sys.setdefaultencoding('utf-8')

logger = get_logger(__file__)

CURRENT_PATH = os.path.dirname(__file__)
if CURRENT_PATH:
    CURRENT_PATH = CURRENT_PATH + "/"

pg = PgUtil()


def add_category_into_resume():
    query = """
            select count(*) from resume
            """
    result = pg.query_one_sql(query)
    total = result['count']
    for id in range(48117,total+1):
        query = """
                select * from resume where id = %s
                """
        values = (id,)
        result = pg.select_sql(query,values)
        text = result['job_title'] if result['job_title'] else '' + result['detail'] if result['detail'] else ''
        _category_des = search_category_type(text.decode())
        if _category_des:
            _category = _category_des.split(':')[1]
            _category_type = _category_des.split(':')[0]

        else:
            _category = u'其他'
            _category_type = u'其他'

        logger.info('update No.%d info'%id)
        _query = """
                 update resume set category = %s, category_type = %s where id =%s
                 """
        _values = (_category,_category_type,id)
        pg.execute_update_sql(_query,_values)

if __name__ == '__main__':
#     search_category_type('''数据通信工程师1.负责客户网络的规划设计，数据的配置，网络的运行维护，设备初验，终验等。
# 2.负责设备问题的跟踪、反馈及疑难问题的处理、技术信息的收集、整理；
# 3.负责新员工和用户的技术培训，配合客户经理做好协调工作；
# 4.提供必要的技术支持，包括技术评价、工程勘察、工程设计等。
# 5.负责项目的招投标工作，包括整体解决方案的拟定、标书应答、讲解与答辩。
# 6.根据客户的需求进行系统概要的设计并编写解决方案。''')
    add_category_into_resume()